cd ../../

gpus=$1
ids_bsz=24

# pretrained_prefix="['sed_student.cnn.cnn.conv3',"` 
#                   `"'sed_student.cnn.cnn.batchnorm3',"` 
#                   `"'sed_student.cnn.cnn.cg3',"` 
#                   `"'sed_student.cnn.cnn.conv4',"` 
#                   `"'sed_student.cnn.cnn.batchnorm4',"` 
#                   `"'sed_student.cnn.cnn.cg4',"` 
#                   `"'sed_student.cnn.cnn.conv5',"` 
#                   `"'sed_student.cnn.cnn.batchnorm5',"` 
#                   `"'sed_student.cnn.cnn.cg5',"`
#                   `"'sed_student.cnn.cnn.conv6',"` 
#                   `"'sed_student.cnn.cnn.batchnorm6',"` 
#                   `"'sed_student.cnn.cnn.cg6']"
# desirable_prefix="['cnn.cnn.conv3',"` 
#                 `"'cnn.cnn.batchnorm3',"` 
#                 `"'cnn.cnn.cg3',"` 
#                 `"'cnn.cnn.conv4',"` 
#                 `"'cnn.cnn.batchnorm4',"` 
#                 `"'cnn.cnn.cg4',"` 
#                 `"'cnn.cnn.conv5',"` 
#                 `"'cnn.cnn.batchnorm5',"` 
#                 `"'cnn.cnn.cg5',"`
#                 `"'cnn.cnn.conv6',"` 
#                 `"'cnn.cnn.batchnorm6',"` 
#                 `"'cnn.cnn.cg6']"
pretrained_prefix="['sed_student.rnn.']"
desirable_prefix="['rnn.']"


echo "Run baseline with first 3-layer CNN initialized by Sup/SelfSL/Finetune results."
python sed_main.py   \
    --gpus "${gpus}," \
    --exp_name 'sed/init/R_BYOLA_ft' \
    --batch_sizes "[0, 12, 12, 24, 0]" \
    --pretrained_ckpt "/home/shaonian/SED/sssl_sed/codes/exp/ckpts/OOD_BYOLA_CRNN_ft.ckpt" \
    --pretrained_prefix ${pretrained_prefix} \
    --desirable_prefix ${desirable_prefix} \
    --freeze_cnn_layer "[-1]" 